//给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。 
//
// 路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。 
//
// 
//
// 示例 1： 
//
// 
//
// 
//输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
//输出：3
//解释：和等于 8 的路径有 3 条，如图所示。
// 
//
// 示例 2： 
//
// 
//输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
//输出：3
// 
//
// 
//
// 提示: 
//
// 
// 二叉树的节点个数的范围是 [0,1000] 
// -10⁹ <= Node.val <= 10⁹ 
// -1000 <= targetSum <= 1000 
// 
// Related Topics 树 深度优先搜索 二叉树 👍 1378 👎 0

package leetcode.editor.cn;

import java.util.HashMap;
import java.util.Map;

class PathSumIii {
    public static void main(String[] args) {
        Solution solution = new PathSumIii().new Solution();
    }
    //leetcode submit region begin(Prohibit modification and deletion)

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    class Solution {
        // 时间复杂度O(n * n)，空间复杂度O(n)
        /*public int pathSum(TreeNode root, int targetSum) {
            if (root == null) return 0;
            int res = 0;
            res += pathSum(root, targetSum);
            res += pathSum(root.left, targetSum);
            res += pathSum(root.right, targetSum);
            return res;
        }


        public int traversal(TreeNode root, int targetSum) {
            if (root == null) return 0;

            int pathCount = 0;
            // 判断本身
            if (root.val == targetSum) pathCount++;
            if (targetSum == 0) pathCount++;
            pathCount += traversal(root.left, targetSum - root.val);
            pathCount += traversal(root.right, targetSum - root.val);
            return pathCount;
        }
*/
        public int pathSum(TreeNode root, int targetSum) {
            Map<Long, Integer> preMap = new HashMap<>();
            // 0的前缀和有一个，当节点本身等于target时也算
            preMap.put(0L, 1);
            int res = dfs(root, preMap, 0, targetSum);
            return res;
        }


        public int dfs(TreeNode root, Map<Long, Integer> preMap, long sum, int targetSum) {
            if (root == null) return 0;
            sum += root.val;
            int pathCount = preMap.getOrDefault(sum - targetSum, 0);
            // 加入前缀和
            preMap.put(sum, preMap.getOrDefault(sum, 0) + 1);
            pathCount += dfs(root.left, preMap, sum, targetSum);
            pathCount += dfs(root.right, preMap, sum, targetSum);
            // 回溯
            preMap.put(sum, preMap.getOrDefault(sum, 0) - 1);
            return pathCount;
        }

    }
//leetcode submit region end(Prohibit modification and deletion)

}
